export const elementMixin = {
  data() {
    return {
      // tree是否全选，复选框默认状态（）
      isSelectAll: false,
      isSelectAll2: false,
      // tree是否展开
      isExpand: true,
      isExpand2: true
    }
  },
  methods: {
    /** 全选/全不选 */
    selectAll(ref = 'myTree', treeDataName = 'treeData') {
      if (this.isSelectAll) {
        // 	设置目前勾选的节点，使用此方法必须设置 node-key 属性
        this.$refs[ref].setCheckedNodes(this[treeDataName])
      } else {
        // 全部不选中
        this.$refs[ref].setCheckedNodes([])
      }
    },
    /** 全选/全不选 */
    selectAll2(ref = 'myTree', treeDataName = 'treeData') {
      if (this.isSelectAll2) {
        // 	设置目前勾选的节点，使用此方法必须设置 node-key 属性
        this.$refs[ref].setCheckedNodes(this[treeDataName])
      } else {
        // 全部不选中
        this.$refs[ref].setCheckedNodes([])
      }
    },
    /** 展开/折叠 */
    expandAll(ref = 'myTree') {
      for (
        var i = 0;
        i < this.$refs[ref].store._getAllNodes().length;
        i++
      ) {
        // 根据isExpand， tree展开或折叠
        this.$refs[ref].store._getAllNodes()[i].expanded = this.isExpand
      }
    },
    /** 展开/折叠 */
    expandAll2(ref = 'myTree') {
      for (
        var i = 0;
        i < this.$refs[ref].store._getAllNodes().length;
        i++
      ) {
        // 根据isExpand， tree展开或折叠
        this.$refs[ref].store._getAllNodes()[i].expanded = this.isExpand2
      }
    },

  }
}
